// 实现异步日志缓冲区
#ifndef __M_BUFF_H__
#define __M_BUFF_H__

#include "util.hpp"
#include <vector>
#include <cassert>

namespace mylog {
    #define DEFAULT_BUFFER_SIZE (1*1024*1024)
    #define THRESHOLD_BUFFER_SIZE (8*1024*1024)
    #define INCREMENT_BUFFER_SIZE (1*1024*1024)
    class Buffer {
        public:
            Buffer():_buffer(DEFAULT_BUFFER_SIZE), _write_idx(0), _reader_idx(0) {}
            // 向缓冲区写入数据
            void push(const char *data, size_t len) {
                // 缓冲区剩余空间不够的情况： 1. 扩容
                // 1. 固定大小，则直接返回
                //if (len > writeAblesize()) return;
                // 2. 动态空间，用于极限性能测试--扩容
                assert(len <= writeAblesize());
                ensureEnoughSize(len);
                // 1. 将数据拷贝进缓冲区
                std::copy(data, data+len, &_buffer[_write_idx]);
                // 2. 将写入位置向后偏移
                _write_idx += len;
            }
            size_t writeAblesize() {
                // 对于扩容思路来说，不存在可写空间大小，因为总是可写
                // 因此这个接口仅仅针对固定大小缓冲区提供
                return (_buffer.size()-_write_idx);
            }
            // 返回可读数据的起始位置
            const char* begin() {
                return &_buffer[_reader_idx];
            }
            // 返回可读数据的长度
            size_t readAbleSize() {
                return (_write_idx-_reader_idx);
            }
            
            void moveReader(size_t len) {
                assert(len <= readAbleSize());
                _reader_idx += len;
            }
            // 重置读写位置，初始化缓冲区
            void reset() {
                _write_idx = 0; // 缓冲区所有空间都是空闲的
                _reader_idx = 0; // 与_write_idx相等表示没有数据可读
            }
            // 对Buffer实现交换操作
            void swap(Buffer &buffer) {
                _buffer.swap(buffer._buffer);
                std::swap(_reader_idx, buffer._reader_idx);
                std::swap(_write_idx, buffer._write_idx);
            }
            // 判断缓冲区是否为空
            bool empty() {
                return (_reader_idx == _write_idx);
            }
        private:
            // 对空间扩容
            void ensureEnoughSize(size_t len) {
                if (len <= writeAblesize()) return ;// 不需要扩容
                size_t new_size = 0;
                if(_buffer.size() < THRESHOLD_BUFFER_SIZE) {
                    new_size = _buffer.size() * 2 + len; // 小于阈值则翻倍增长
                }else {
                    new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len; // 否则线性增长
                }
                _buffer.resize(new_size);
            }
            // 对读写指针进行向后偏移操作
            void moveWriter(size_t len) {
                assert((len + _write_idx) <= _buffer.size());
                _write_idx += len;
            }
        private:
            std::vector<char> _buffer;
            size_t _reader_idx; // 当前可读数据的指针--本质是下标
            size_t _write_idx; // 当前可写数据的指针
    };
}

#endif